#!/bin/sh

#
# Test U-boot support for setting variables
#

. "$(cd "$(dirname "$0")" && pwd)/common.sh"


# Create a starter uboot environment block
# This is created by running:
#
# $ printf "\n" | mkenvimage -s "512" - | base64
base64_decode >$WORK/starting-uboot-env.img <<EOF
gmEfOQD/////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////8=
EOF

cat >$CONFIG <<EOF
define(UBOOT_ENV_OFFSET, 32)

file-resource uboot-env.img {
    host-path = "$WORK/starting-uboot-env.img"
}

uboot-environment uboot-env {
    block-offset = \${UBOOT_ENV_OFFSET}
    block-count = 1
}

task complete {
    on-resource uboot-env.img {
        raw_write(\${UBOOT_ENV_OFFSET})
    }

    on-finish {
       uboot_setenv(uboot-env, "firstvar", "hello")
    }
}
EOF

# Create the expected result from setting the variables.
# To recreate, run:
#
# printf "firstvar=hello" | mkenvimage -s "512" - | base64
#
base64_decode >$WORK/expected-env.img <<EOF
+MFCCmZpcnN0dmFyPWhlbGxvAAD/////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////8=
EOF

# Create the firmware file, then "burn it"
$FWUP_CREATE -c -f $CONFIG -o $FWFILE
$FWUP_APPLY -a -d $IMGFILE -i $FWFILE -t complete

# The firmware file is equivalent to the following dd call
dd if=$IMGFILE of=$WORK/actual-env.img skip=32 count=1 2>/dev/null
cmp $WORK/expected-env.img $WORK/actual-env.img

# Check that the verify logic works on this file
$FWUP_VERIFY -V -i $FWFILE
